Crypto-js equivalent function in python